package com.djbx.points.common.core.utils;

import com.baomidou.mybatisplus.core.metadata.IPage;
import org.springframework.beans.BeanUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.function.Supplier;

/**
 * @author Administrator
 * @date 2021/11/23 9:57
 */
public class BeanListUtils extends BeanUtils {

    /**
     * 对象拷贝
     * @param source 源对象
     * @param target 目标类
     * @param <S> source类型
     * @param <T> target类型
     * @return
     */
    public static <S, T> T copyProperties(S source, Supplier<T> target) {
        T t = null;
        if(source != null){
            t = target.get();
            copyProperties(source, t);
        }
        return t;
    }

    /**
     * 集合数据的拷贝
     * @param sources: 数据源类
     * @param target: 目标类::new(eg: UserVO::new)
     * @return
     */
    public static <S, T> List<T> copyListProperties(List<S> sources, Supplier<T> target) {
        return copyListProperties(sources, target, null);
    }

    public static <S, T> PageList<T> copyPageProperties(IPage<S> sourcePage, Supplier<T> target) {
        PageList<T> targetPage = new PageList<T>();
        targetPage.setCurrPage(sourcePage.getCurrent());
        targetPage.setPageSize(sourcePage.getSize());
        targetPage.setTotalPage(sourcePage.getPages());
        targetPage.setTotalCount(sourcePage.getTotal());
        List<S> sources = sourcePage.getRecords();
        targetPage.setList(copyListProperties(sources, target, null));
        return targetPage;
    }

    /**
     * 带回调函数的集合数据的拷贝（可自定义字段拷贝规则）
     * @param sources: 数据源类
     * @param target: 目标类::new(eg: UserVO::new)
     * @param callBack: 回调函数
     * @return
     */
    public static <S, T> List<T> copyListProperties(List<S> sources, Supplier<T> target, BeanListUtilCallBack<S, T> callBack) {
        List<T> list = new ArrayList<>(sources.size());
        for (S source : sources) {
            T t = target.get();
            copyProperties(source, t);
            list.add(t);
            if (callBack != null) {
                // 回调
                callBack.callBack(source, t);
            }
        }
        return list;
    }
}
